from django.urls import path, include
from rest_framework.routers import DefaultRouter
from .views import UserViewSet, store_files, get_chat_chain, store_file_ollama, get_chat_chain_ollama
from .views import hello_world
from .views import start_prompt

router = DefaultRouter()
router.register(r'messages', UserViewSet)

urlpatterns = [
    path('', include(router.urls)),
    path('simple/', hello_world),
    path('start_prompt/', start_prompt),
    path('store_files/', store_files),
    path('get_chat_chain/', get_chat_chain),
    path('store_file_ollama/', store_file_ollama),
    path('get_chat_chain_ollama/', get_chat_chain_ollama),
]
